import logging
import os
from logging import handlers


class Logger(object):
    """
    日志模块
        输出log到控制台以及将日志写入log文件。
        保存2种类型的log， all.log 保存debug, info, warning, critical 信息， error.log则只保存error信息，同时按照时间自动分割日志文件。
    """
    level_relations = {
        'debug': logging.DEBUG,
        'info': logging.INFO,
        'warning': logging.WARNING,
        'error': logging.ERROR,
        'crit': logging.CRITICAL
    }  # 日志级别关系映射

    def __init__(self, filename, level='info', when='midnight', interval=7, back_count=5,
                 fmt='%(asctime)s - %(levelname)s - %(message)s'):
        log_dir = 'logs'  # 日志统一存放文件夹
        log_dir_path = os.path.join(os.getcwd(), log_dir)  # 日志统一存放完整路径:默认当前路径下的logs文件夹
        if not os.path.exists(log_dir_path):  # 日志统一存放路径不存在，则创建该路径
            os.makedirs(log_dir_path)
        self.logger = logging.getLogger(log_dir_path + '/' + filename)

        format_str = logging.Formatter(fmt)  # 设置日志格式
        self.logger.setLevel(self.level_relations.get(level))  # 设置日志级别
        sh = logging.StreamHandler()  # 往屏幕上输出
        sh.setFormatter(format_str)  # 设置屏幕上显示的格式
        th = handlers.TimedRotatingFileHandler(filename=filename, when=when, backupCount=back_count, encoding='utf-8')
        # 实例化 TimedRotatingFileHandler
        # interval 是when的间隔，back_count 是备份文件的个数，如果超过这个个数，就会自动删除，when是间隔的时间单位，单位有以下几种：
        # S 秒
        # M 分
        # H 小时、
        # D 天 按当前时间累加一天，可能有坑
        # midnight 每天凌晨
        # W 每星期 格式为：w0表示星期1
        th.setFormatter(format_str)  # 设置文件里写入的格式
        self.logger.addHandler(sh)  # 把对象加到logger里
        self.logger.addHandler(th)


if __name__ == '__main__':
    # 调用本模块的用法
    # log = Logger('all.log', level='debug')
    logger = Logger('logs/xxx.log', level='debug').logger
    logger.debug('debug')
    logger.info('info')
    logger.warning('警告')
    logger.error('报错')
    logger.critical('严重')
    Logger('error.log', level='error').logger.error('error')
